creating new category column in python pandas based on existing column